Revert "Revert "removed .py extensions""

This reverts commit d31c997f84cde56b4f421e396cef308c9d7d19e7.