Tidy up problem1