finished first challenge, but the solution could be better