Moved add_data functionality to initialize method