Added case for linear kernel