Refactor code to use python case.