Added more to the final code sample