Added current cell to reward equation