Coorect code after test menu and file system work