Part 1 almost working