More docs, api change to attention api