worked now add class