feature(ai-hw1): adding part of the work