some quick cleanup on the OutputLayer