Improves on code