Hopefully finished python script