Made some additional changes to the code.