Added more reward functionality