Works fine

Need better design and some more features