Modified head() to take a parameter n for the number of rows to return. Cleaned up unused imports and updated the readme.