Added small changes