Working but needs tweaking.