Phase Retrieval Python