More sensible output layer