Use class attribute
Remove rainbow mode