Code added from python class