query method refactored

Hej, masz tu pull request, ten kod powinien działać :)

Zrobiłem kilka błędów drobnych, poprawiłem w kilka minut ;)