Improved readme with code example