Done task1. Need to refactor code.