Added variables to code