Completed if/else statement. Should fix code to use 4 space tabs at some point too.