Trains weights pretty well with simplified logic