update solution to solo challenge