Improve assign example