Small changes to get_code_samples