lab 5 almost done, I just need to add the ui.