Add missing functionality in GUI