Adding main procedure to tests file