updating solution to solo challenge