added more code

second commit