Added content to complete the assignment