added more code & updated readme file