Final project for online python course