added part of hw#4