Added suggestion for improvement