Nice Angles easy challenge solved in java and python