Improve python codes
Remove python codes