Some small changes made