hw-4: fixed syntax and includes in test program