fixes issue #1

added stuff on line 5