Atlas - Improve and Repeat : Python