Added branch checkout and tests