minor editing in code