Added TODO comments and few lines of documentation where I could.