code style

better code