Add gui support