Code cleanup. Thanks @NetzwergX