Fixed snippet for python class