I need to apply this on my class