MAde some small changes