Fixed a bug with problem 5.