Minor modification to code

Done some minor modifications for more precision