See todo(s) in code

Hi.
I've found some critical issues with your code.
You could fix them, if you want to learn as much as possible from this task.