Made small changes