Cleaned up example code in readme to match actual example code