Fix code to make angles nice