Adding initial problem 4 implementation - not working.