clean up a little

clean up a little