Solve problem #2 in Julia