Small changes for code