Fix draw_board to return instead of print