Add the main tests for the third homework