Comments updated and .py extension added to indicate file type