used classes instead of hard code