Code added

Test file deleted, main code added