TODO

Make it nicer asap.