Started work on UI. Disregard it for now