python factorial program with function