started on python solutions (1, 6)