problem 4
playing around with modules