small fix..
small changes..