Cleans up example and code