Finally figured our problem 4