little better code