Fixed code samples a bit more