Make code snippet match screenshots