Solution needs work