Everything is currently working, messing around with more UI stuff