updating to handle =>, should work now