Working on NN, added rotation and translation