Challenge - Refactor Perl into Python