Wrote tests for Problem
formatted tests