funkcia za trest