completed initial code for solo challenge