Library fix and annotation fix

[neeleshverma]

numpy should be used instead of math for better precision and a small annotation fix.

[neeleshverma]

The "params_only" parameter is already there in the function. I simply added the type (bool) to make the function declaration clear.