cleaned up plot formatting for nonlin model demo