Refaktor answer

prerobena funkcia answer(martin)