Adds cleaned up version of the code