better code

tiny fixes