Dont hardcode class names