Ha, right as I pushed the last commit, I saw I could actually do this in a better way

Signed-off-by: Daktyl198 <rootme198@hotmail.com>