added initial code for 5-3 task