Close, but still have shitty logic.