Fixed it to be kind of working