Cleaned up test and removed some prints.