Still trying to get it working perfectly