rewrote problem 2 - passing