Fixing weight issue and adding simple tests for get_best_move