predict method + small test