Fixed the bug in the GA. Now it trains better, although it seems to get suck in local optima around 3-5