Some logic done
Good code