not passing more tests but I think i know how to fix it