#7 -- WIP

Attempted adding UI