Added the def function allowing users to define variables