Refactor

Signed-off-by: LordIllyohs <lordillyohs@gmail.com>