Reorder code sample for better flow