Add controller for now
Fix resizing issue with directory