Must get better at python, and testing