Code cleanup

Signed-off-by: Swizzy <Swizzy@xeupd.com>