cleaned up py file