Made function return consistent lists for each case
Created main function, and updated README.