Added problem 4

Added problem 4 and created a variable for the results of the first
part.