add problem statement to #4
got rid of some comments