add code for the problems