This is just a rough draft. Only trained on 2000 steps but the results seem to be pretty solid already. Will upload a more trained version later