Added problem{X}Answer to run the code to sort the problem and cleaned up a bit