finished up ui stuff