Starting to fix reward stuff