fixed up a little