Finished happiest_state.py for Problem 5