Added main file for testing