[TASK] Enhanced demo to show off postLayout method