Revert "removed .py extensions"

This reverts commit 51c3e9aeaf01a4679ab2c4080ecf834de06f15ee.