Overloaded the assume and nondet function for a few types