Added description and fixed main