Better main function
Tidying up README