Example clean up
Fix GPU example in docs