Calculate Distance easy challenge solved in python