Updated script to new class location