Screwed around with heuristics

Added weight for corner and edge. Sort of works.