Updated to work correctly :)