Works! but is buggy