lab6 added pred and stuff