Code clean for screenshot