Just a little cleanup to start

I've never written python so it's a little slow going, but I did manage to get the app running locally :)