Updated file to reflect the new Class