Added a note...
Organized the code, minor fix