Two Dimensional Transistors With Non-Volatile Functionalities