V0.2.1 : Small tweaks