Cole vows to improve