Updated UI, cleaned up repo