adding some lines I missed