Figure Method In Geeksforgeeks In Python