commented a todo
fixed bug in end{figure}