fixed more UI stuff