Selection UI almost done