Added check for pro class