small changes, not working