Adding code to find maximum margin linear separator