Move NN layers into a separate file