updated to class selector to reflect latest changes
removed pry