made code pretty: YAY!