Added few classes for different states