Update rot.py

made it able to handle floating point numbers