Added classes python file