Clearer example in readme file with imports etc.