Refactoring.

Signed-off-by: katlogic <kat@lua.cz>