将原有代码改写为类