Fix my Code again (I've use Python too much)