added another method to problem 4 and a utility function to utils.py