updated my solution to first challenge