fixed it up a little