Cleaned up reward code. Don't think it was working as intended.