Definitely needs a lot, if I have time I will improve this I really enjoyed developing it.