Made code sample simpler