Created test for the initial_data method
Final changes