Cleans up the code for example