cleaning up code in problems.jl