We don't need the 'loss' variable