Creating a fully connected model and train