Added extra bug fix to hide for medium class