just like python
succint way of solving this. im proud