Polished example code in readme