Added variables to the custom class