Added final output layer