Fixed rotation bug for assignment 4

need to rebase proj again