amend to make binary prediction + metrics