Solve problem #1 in Julia