hw-2: Made changes in order to make code compile.