Removed all the modules but nn