First go solution, ripped from python to get used to go basics