More code but equals for answers is broken