updated ui some to look better