Add missing code, update README