Fixes #10 Consolidated ui update stuff into one method