Some small changes added