fixed a typo and added params to the function get_yprime