Worked on another code eval problem