Random shape generator in python :)