Back to masked inputs (also fixed bug with no reward selected)