Tried to add  click events and realised how shit my code has got... refactoring...