Ok so, remove hardcoded class names in this file can be a good start ^^