Minor docs & remove print statements