Add return for pretty print