Prob 04

I am not really comfortable using float here, but I'm not sure whether it shouldn't be done.

Signed-off-by: aaruni96 <aaruni.kaushik@gmail.com>