Add support of extra styling class on collapse event