Added a wretchedly ugly UI! Roughly what I want though.