Set up to work with right directory