add model directory and make UI pretty