Add additional css logic