After doing a little more research, it seems as though doing a full swap wouldn't be worth it, considering from what I understand the baja/STI motors are quite similar. The differences I found between the two are the turbo itself, intercooler, up-pipe and ECU. It might be just worth my time just to do simple upgrades such as these. Still open for advice and opinions though.