Initial solution (the correct type this time)