Rest of the code was added