prettied up some code