Flatten layers dense layers and softmax