Changed checks file to add more class tags