correct solution

correct solution