Cleaned up the code

Signed-off-by: bluemonster122