Urgh I haven't touched Python in too long. Herp derp ifs need statements derp