small fiz in zad2 - return list added