Minor updates in code