small changes for improvments