Fix if statement. Fix back()