Fixed code sample on readme