refactor code to OO style

求真谛老师指教。