bug fixes + ui