bugfixes, updating code..