Add browser button and clean up excess imports