add main tests