Edited a few lines of code