Not the best at python :)