Kind of finished.. Not working 100% properly