Add more comments within functions