Will implement with the image later as it's a headache now