changing a few things to make it work