Added few more lines into programm