hw-2: updated script