Update happiest_state.py