FINALLLLY!!! Its a bit hardcoded but it works.