Removed some prints. Seems to work now.