Basic weight initialisation works