Added new class to element and used that instead