change the answer logic a bit