Add solution and move problem desc
Rewrite as class