Changed back to class not id