need to turn script into usable class