added some small changes
added some small changes again.