changed line 5 and 6