feature(ai-hw1): done some more exersises