agrege el codigo que llevo hecho en python