Just some small changes. Removing prints etc