code by python