updated code and tests but not working still