work on #3, lets do this with python