Homework 3 fix Kaloyan Valkov