Udemy implementation complete - need to fix image bug