Solved the first couple of problems in Go