Small changes

Added some small change