Add shell for problem 4
Solve problem 4