School work involving python coding