Adding my solution for #2