Updated to something similar to this, thanks!