refactored problem 1 to have a more elegant format using "solve_problem" method
refactored problem 2
need this __init__!!!
updated tests for python problems 1 & 2