add function to switch between nornu and rnu