Few comments and updated readme