First usable version. Added some basic UI