Yeah I'm not too sure on this part. Seems a bit hacky to me.