Sorting Lego sucks, so here's an AI that does it for you