How To Put Code In A Class Python