added more code

adding new code from line # 11 to 14