Added some UI, might need to rewrite some of it