better code style

colinkho