Should add support for checking out a branch