Add a script to use class in python