Works, just need to mess around with some minor GUI stuff