修改HW1。使用class去編寫