Updated a few parts of the code