Updated writeup and cleaned up UI